/*
 * @lc app=leetcode.cn id=496 lang=cpp
 *
 * [496] 下一个更大元素 I
 */

// @lc code=start
class Solution
{
public:
  vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2)
  {
    int n = nums2.size();
    unordered_map<int, int> m;
    for (int i = 0; i < n; ++i)
      m[nums2[i]] = i;
    for (int& num : nums1) {
      int idx = m[num];
      while (++idx < n && nums2[idx] < num)
        ;
      num = idx == n ? -1 : nums2[idx];
    }
    return nums1;
  }
};
// @lc code=end
